#pragma once

#include <iostream>
#include <vector>
#include <queue>
#include "thread.hpp"
#include "Log.hpp"
#include "Mutex.hpp"
#include "Cond.hpp"

namespace ThreadPoolModule
{
    using namespace MutexModule;
    using namespace CondModule;
    using namespace LogModule;
    using namespace ThreadModlue;

    // 线程个数
    static const int gnum = 5;
    template <class T>
    class ThreadPool
    {
    private:
        void WakeUpALLThread()
        {
            LockGuard lockguard(_mutex);
            if (_sleepernum)
                _cond.Broadcast();
            LOG(LogLevel::INFO) << "唤醒所有休眠线程";
        }
        void WakeUpOne()
        {
            _cond.Signal();
            LOG(LogLevel::INFO) << "唤醒一个休眠线程";
        }
        // 启动线程
        void Start()
        {
            // 线程池正在运行中
            if (_isrunning)
                return;
            // 启动线程池
            _isrunning = true;
            for (auto &thread : _threads)
            {
                thread.Start();
                // 打印一条日志
                LOG(LogLevel::INFO) << "线程启动";
            }
        }
        ThreadPool(const int &num = gnum)
            : _num(num), _sleepernum(0), _isrunning(false)
        {
            // 创建线程
            for (int i = 0; i < _num; i++)
            {
                // 传入回调函数
                _threads.emplace_back(
                    [this]()
                    {
                        HandlerTask();
                    });
            }
        }

    public:
        // 获取单例指针
        static ThreadPool<T> *GetInstance()
        {
            if (inc == nullptr)
            {
                LockGuard lockguard(_lock);
                LOG(LogLevel::INFO)<<"获取单例...";
                if (inc == nullptr)
                {
                    LOG(LogLevel::INFO)<<"首次使用单例，创建单例...";
                    inc = new ThreadPool<T>();
                    inc->Start();
                }
            }
            return inc;
        }
        // 线程暂停
        void Stop()
        {
            // 线程池已经结束
            if (!_isrunning)
                return;
            _isrunning = false;
            // 暂停线程池，需要等待任务队列中的任务都被执行完
            // 唤醒所有线程
            WakeUpALLThread();
        }
        void HandlerTask()
        {
            // 执行任务
            // 获取对应线程名
            char name[64];
            pthread_getname_np(pthread_self(), name, sizeof(name));
            while (true)
            {
                T t;
                {
                    LockGuard lockguard(_mutex);
                    // 任务队列如果为空，进行等待
                    // 线程池暂停的条件：
                    // 1，_isrunning=false && 2,任务队列为空
                    // 唤醒休眠中的线程执行任务
                    // 线程池正在运行并且队列为空，就会继续休眠
                    while (_isrunning && _taskq.empty())
                    {
                        _sleepernum++;
                        _cond.Wait(_mutex);
                        _sleepernum--;
                    }
                    // 没有任务并且线程池状态为暂停，跳出循环
                    if (!_isrunning && _taskq.empty())
                    {
                        LOG(LogLevel::INFO) << "线程暂停";
                        break;
                    }
                    // 线程池要暂停或者队列不为空，线程就会执行对应任务
                    // 取任务
                    t = _taskq.front();
                    _taskq.pop();
                }
                // 执行任务
                // t();//不需要在临界区中执行
            }
        }
        // 任务入队列
        void Enque(const T &in)
        {
            LockGuard lockguard(_mutex);
            _taskq.push(in);
            if (_threads.size() == _sleepernum)
                WakeUpOne();
        }
        // 等待所有线程退出
        void Join()
        {
            for (auto &thread : _threads)
            {
                thread.Join();
                LOG(LogLevel::INFO) << "线程退出";
            }
        }
        ~ThreadPool()
        {
        }

    private:
        std::vector<Thread> _threads; // 线程池
        std::queue<T> _taskq;         // 任务队列
        int _num;                     // 线程的个数
        bool _isrunning;              // 线程是否在运行
        int _sleepernum;              // 处于休眠状态的线程个数
        Mutex _mutex;
        Cond _cond;
        // 单例模式
        static ThreadPool<T> *inc; // 单例指针
        static Mutex _lock;
    };
    // 静态成员变量在类外初始化
    template <class T>
    ThreadPool<T> *ThreadPool<T>::inc = nullptr;
    template <class T>
    Mutex ThreadPool<T>::_lock;
}